package com.zhijian.medical.service.impl;

import com.zhijian.medical.dao.local.TTcCityMapper;
import com.zhijian.medical.entity.po.TTcCity;
import com.zhijian.medical.entity.vo.request.TTcCitySearchVo;
import com.zhijian.medical.entity.vo.response.TTcCityVo;
import com.zhijian.medical.service.TTcCityService;
import com.zhijian.medical.util.Page;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Objects;

/**
 * 城市业务处理层
 *
 * @author HCG
 * @date 2021-10-25 21:54:16
 * @version: 1.0
 */
@Slf4j
@Service
public class TTcCityServiceImpl implements TTcCityService {

    @Autowired
    private TTcCityMapper mapper;

    @Override
    public TTcCity selectById(Long id) {
        return mapper.selectByPrimaryKey(id);
    }


    @Override
    public Integer count(TTcCitySearchVo tcCitySearchVo) {
        return mapper.count(tcCitySearchVo);
    }

    @Override
    public List<TTcCityVo> selectList(TTcCitySearchVo tcCitySearchVo) {
        return mapper.selectList(tcCitySearchVo);
    }

    @Override
    public Page<TTcCityVo> selectPage(TTcCitySearchVo tcCitySearchVo) {
        //先直接查总数
        int count = count(tcCitySearchVo);
        //分页
        Page<TTcCityVo> page = new Page<>(true, count, tcCitySearchVo.getPageNum(), tcCitySearchVo.getPageSize());
        tcCitySearchVo.setStartIndex(page.getStartIndex());
        page.setPage(true);
        page.setTotalCount(count);
        List<TTcCityVo> responseVos = this.selectList(tcCitySearchVo);
        page.setData(responseVos);
        return page;
    }

    @Override
    public List<TTcCityVo> renderTreeList() {
        List<TTcCityVo> tTcCityVos = this.selectList(TTcCitySearchVo.builder().build());
        Collections.reverse(tTcCityVos);
        return createTree(tTcCityVos, 1L);
    }

    /**
     * 生成树的方法
     */
    private List<TTcCityVo> createTree(List<TTcCityVo> list, Long pid) {
        try {
            //根节点
            List<TTcCityVo> tempList = new ArrayList<>();
            for (TTcCityVo tTcCityVo : list) {
                //父节点是0的，为根节点。
                if (Objects.equals(tTcCityVo.getParentId(), pid)) {
                    tempList.add(tTcCityVo);
                }
            }
            //为根id设置子childList是递归调用的
            for (TTcCityVo tTcCityVo : tempList) {
                /* 获取根节点下的所有子节点 使用getChild方法*/
                List<TTcCityVo> childList = setChild(tTcCityVo.getId(), list);
                //给根节点设置子节点
                tTcCityVo.setChildren(childList);
            }
            // 输出构建好的菜单数据。
            return tempList;
        } catch (Exception e) {
            return new ArrayList<>();
        }
    }

    /**
     * 获取子节点
     *
     * @param id   父节点id
     * @param list 所有节点列表
     * @return 每个根节点下，所有子节点列表
     */
    private List<TTcCityVo> setChild(Long id, List<TTcCityVo> list) {
        //子节点
        List<TTcCityVo> childList = new ArrayList<>();
        for (TTcCityVo tTcCityVo : list) {
            // 遍历所有节点，将所有子节点的父id与传过来的根节点的id比较
            //相等说明：为该根节点的子节点。
            if (tTcCityVo.getParentId().equals(id)) {
                childList.add(tTcCityVo);
            }
        }
        //递归
        for (TTcCityVo tTcCityVo : childList) {
            tTcCityVo.setChildren(setChild(tTcCityVo.getId(), list));
        }
        //如果节点下没有子节点，返回一个空List（递归退出）
        if (childList.isEmpty()) {
            return new ArrayList<>();
        }
        return childList;
    }

}